from django.shortcuts import render
from django.contrib import admin
from django.contrib.auth import get_user_model
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.utils.translation import gettext, gettext_lazy as _
from django.contrib.auth.hashers import make_password, check_password

User = get_user_model()

# admin.site.unregister(User)

"""
    DJango后台管理系统作用
    1.通过@admin.register("需要管理的模型")或则admin.site.register("...")去注册进后台管理系统界面，可以在上面对模型进行更改
"""


# 通过装饰器注册数据模型User给admin界面去管理
@admin.register(User)
class UserAdmin(BaseUserAdmin):
    # 定义显示的字段
    list_display = ('username', 'is_active', 'belong_groups')

    # 编辑资料的时候显示的字段
    fieldsets = (
        (None, {'fields': ('username', 'password')}),
        (_('Permissions'), {
            'fields': ('is_active', 'is_staff', 'is_superuser', 'groups',),
        }),
    )
    # 新增用户需要填写的字段
    add_fieldsets = (
        (
            None,
            {
                'classes': ('wide',),
                'fields': ('username', 'password1', 'password2', 'is_active', 'is_staff', 'groups'),
            }
        ),
    )

    filter_horizontal = ('groups',)

    def belong_groups(self, obj):
        return ", ".join([g.name for g in obj.groups.all()])


    belong_groups.short_description = '所属分组'
